package com.gitee.zeric.codec.util;

import cn.hutool.core.codec.Base64;
import cn.hutool.core.img.ImgUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.extra.qrcode.QrCodeUtil;
import org.krysalis.barcode4j.impl.code39.Code39Bean;
import org.krysalis.barcode4j.output.bitmap.BitmapCanvasProvider;
import org.krysalis.barcode4j.tools.UnitConv;

import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;


/**
 * 条形码工具类
 *
 * @author zj
 * @date 2021/12/14
 */
public class BarcodeUtil {

    /**
     * 生成文件
     *
     * @param msg
     * @param path
     * @return
     */
//    public static File generateFile(String msg, String path) {
//        File file = new File(path);
//        try {
//            generate(msg, new FileOutputStream(file));
//        } catch (FileNotFoundException e) {
//            throw new RuntimeException(e);
//        }
//        return file;
//    }

    /**
     * 生成字节
     *
     * @param msg
     * @return
     */
    public static byte[] generate(String msg,String format,int dpi,int height) {
        ByteArrayOutputStream ous = new ByteArrayOutputStream();
        generate(msg, format,dpi,height,ous);
        return ous.toByteArray();
    }

    /**
     * 生成到流
     *
     * @param msg
     * @param ous
     */
    public static void generate(String msg,String format,int dpi,int height, OutputStream ous) {
        if (StrUtil.isEmpty(msg) || ous == null) {
            return;
        }
        Code39Bean bean = new Code39Bean();
        // module宽度
        final double moduleWidth = UnitConv.in2mm(1.0f / dpi);
        // 配置对象
        bean.setModuleWidth(moduleWidth);
        bean.setWideFactor(3);
        bean.setBarHeight(height);
        bean.doQuietZone(false);
        try {
            // 输出到流
            BitmapCanvasProvider canvas = new BitmapCanvasProvider(ous, format, dpi,
                    BufferedImage.TYPE_BYTE_BINARY, false, 0);
            // 生成条形码
            bean.generateBarcode(canvas, msg);
            // 结束绘制
            canvas.finish();
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    /**
     * 生成条形码
     * @author zj
     * @date 2022/4/29 16:14
     * @param msg
     * @param format
     * @param dpi
     * @param height
     * @return java.lang.String
     */
    public static String generateBarCodeBase64(String msg,String format,int dpi,int height){
        try(ByteArrayOutputStream os = new ByteArrayOutputStream()){
            generate(msg,format,dpi,height,os);
            return Base64.encode(os.toByteArray());
        } catch (IOException e) {
            e.printStackTrace();
        }
        return "";
    }

    /**
     * 生成二维码
     * @author zj
     * @date 2022/4/29 16:15
     * @param msg
     * @param width
     * @param height
     * @return java.lang.String
     */
    public static String generateQrCodeBase64(String msg,int width,int height){
        try(ByteArrayOutputStream os = new ByteArrayOutputStream()){
            QrCodeUtil.generate(msg, width, height, ImgUtil.IMAGE_TYPE_PNG, os);
            return Base64.encode(os.toByteArray());
        } catch (IOException e) {
            e.printStackTrace();
        }
        return "";
    }

}
